from src import heuristics, problems, measure, astar_solution
import matplotlib.pyplot as     plt

n = 10

prob_sets = problems.incrementing_dirt_sets(n, 3)

print prob_sets

algo = astar_solution.astar_solution(heuristics.man_min())
man_min_data = measure.measureTime(prob_sets, algo)

algo = astar_solution.astar_solution(heuristics.fast_manhattan_dirt())
fast_data = measure.measureTime(prob_sets, algo)

algo = astar_solution.astar_solution(heuristics.manhattan_dirt())
manhatan_data = measure.measureTime(prob_sets, algo)

algo = astar_solution.astar_solution(heuristics.scaled_manhattan_dirt(2.5))
s_manhatan_data = measure.measureTime(prob_sets, algo)

plt.legend(['r--'],['scaled-2.5'])
plt.plot(range(n), s_manhatan_data, 'r--', range(n), fast_data, 'b--', range(n), manhatan_data, 'g--')
plt.show()
